from ultralytics import YOLO

import Config


def load_fire_smoke_model():
    model = YOLO(Config.fire_smoke_model)
    return model


def load_sea_ships_model():
    model = YOLO(Config.sea_ship_model)
    return model


def load_hat_model():
    model = YOLO(Config.hat_model)
    return model


def load_person_model():
    model = YOLO(Config.person_model)
    return model
